package com.libre.qr;


import org.krysalis.barcode4j.impl.code39.Code39Bean;
import org.krysalis.barcode4j.output.bitmap.BitmapCanvasProvider;
import org.krysalis.barcode4j.tools.UnitConv;
import org.springframework.util.StringUtils;

import java.awt.image.BufferedImage;
import java.io.*;

/**
 * Copyright: Copyright (c) 2020 XW
 *  BarcodeUtil条形码生成工具类
 * <p> Description:
 *
 * @author fucc
 * @version 1.0.0 2021/4/14:11:13
 * @since 1.0.0
 */
public class BarcodeUtil {

    /**
     * 生成文件到本地路径
     * 生成Code-39条形码
     * @param msg: 条码内容
     * @param path: 存储路径加文件名称
     * @param codeType :生成的条形码类型  1-code39  2-code128
     * @return
     */
    public static File generateFile(String msg, String path) {
        File file = new File(path);
        try {
            generate(msg, new FileOutputStream(file));
        } catch (FileNotFoundException e) {
            throw new RuntimeException(e);
        }
        return file;
    }

    /**
     * 生成字节
     *
     * @param msg
     * @return
     */
    public static byte[] generate(String msg) {
        ByteArrayOutputStream ous = new ByteArrayOutputStream();
        generate(msg, ous);
        return ous.toByteArray();
    }

    /**
     * 生成到流
     *
     * @param msg
     * @param ous
     */
    private static void generate(String msg, OutputStream ous) {
        if (StringUtils.isEmpty(msg) || ous == null) {
            return;
        }

        Code39Bean bean = new Code39Bean();

        // 精细度
        final int dpi = 150;
        // module宽度
        final double moduleWidth = UnitConv.in2mm(1.0f / dpi);

        // 配置对象
        bean.setModuleWidth(moduleWidth);
        bean.setWideFactor(3);
        bean.doQuietZone(false);

        String format = "image/png";
        try {

            // 输出到流
            BitmapCanvasProvider canvas = new BitmapCanvasProvider(ous, format, dpi,
                    BufferedImage.TYPE_BYTE_BINARY, false, 0);

            // 生成条形码
            bean.generateBarcode(canvas, msg);

            // 结束绘制
            canvas.finish();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    private static String path = System.getProperty("user.dir") + "\\src\\main\\resources\\qr\\";
    public static void main(String[] args) throws Exception {
        String msg = "210414000002";
        String path1 = path + "code128.png";
        BarcodeUtil.generateFile(msg, path1);
    }
}
